how to check armstrong number in python